package com.outsider.util.pageprocessor;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

import com.outsider.movie.dao.Movie;

import us.codecraft.webmagic.Page;
import us.codecraft.webmagic.Site;
import us.codecraft.webmagic.processor.PageProcessor;

public class DyttPageprocessor implements PageProcessor{
    // 部分一：抓取网站的相关配置，包括编码、抓取间隔、重试次数等
    private Site site = Site.me().setRetryTimes(3).setSleepTime(1000);
    // process是定制爬虫逻辑的核心接口，在这里编写抽取逻辑
    public void process(Page page) {
		// 部分二：定义如何抽取页面信息，并保存下来
		if (page.getUrl().toString().equals("http://www.dytt8.net/html/gndy/dyzz/index.html"))
			page.addTargetRequests(page.getHtml().xpath("//*[@id=\"header\"]/div/div[3]/div[3]/div[2]/div[2]").links()
					.regex("http://www.dytt8.net/html/gndy/dyzz/[0-9]*/[0-9]*.html").all());
		else {
			Integer dyttID=Integer.parseInt(page.getUrl().toString().split("\\.|/")[9]);
			if (Movie.dao.isExist(dyttID)) return;
			String temp=page.getHtml()
					.xpath("//*[@id=\"header\"]/div/div[3]/div[3]/div[2]/div[2]/div[1]/h1/font/text()").toString();
			Pattern p = Pattern.compile("《(.*)》");
			Matcher m = p.matcher(temp); 
			m.find();
			page.putField("title", m.group(1));
			page.putField("url", page.getHtml().xpath("//*[@id=\"Zoom\"]/span/table/tbody/tr/td/a/text()").toString());
			page.putField("dyttID", dyttID);
			page.putField("date", page.getUrl().toString().split("\\.|/")[8]);
		}
    }

    public Site getSite() {
        return site;
    }

}
